import argparse
from trainer import *

parser = argparse.ArgumentParser()
parser.add_argument('--local_rank', type=int, default=-1)
parser.add_argument('--config', type=str, default='./config/train_ADK20.yaml')
args = parser.parse_args()
trainer = Trainer(args)
trainer.train()